package cn.iocoder.yudao.module.written.exam.enums.llm;

import lombok.AllArgsConstructor;
import lombok.Getter;

/**
 * 大模型消息类型枚举
 *
 * @author 芋道源码
 */
@Getter
@AllArgsConstructor
public enum LlmMessageTypeEnum {

    USER("user", "用户消息"),
    ASSISTANT("assistant", "助手消息"),
    SYSTEM("system", "系统消息");

    private final String type;
    private final String name;

    /**
     * 判断是否为用户消息
     */
    public static boolean isUser(String type) {
        return USER.getType().equals(type);
    }

    /**
     * 判断是否为助手消息
     */
    public static boolean isAssistant(String type) {
        return ASSISTANT.getType().equals(type);
    }

    /**
     * 判断是否为系统消息
     */
    public static boolean isSystem(String type) {
        return SYSTEM.getType().equals(type);
    }

} 